# -*- coding: utf-8 -*-
import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':

    # Load a model
    model = YOLO(model=r'./runs/detect/train/weights/best.pt')  # segment   / classify  /  detect
    model.predict(source=r'./paper_data/Test',
                  save=True,
                  show=False,
                  device = '0'
                  )
